'''
Author: didiplus
Date: 2024-09-24 16:12:37
LastEditTime: 2024-09-24 16:20:31
LastEditors: didiplus
Description: 
FilePath: \pythonscript\python\趣学Python算法100例\11、三色球\demo.py
'''
def count_color_combinations(red, white, black, total):
    """
    计算给定颜色数量下能组成的颜色组合。

    该函数根据给定的红、白、黑三种颜色的数量，计算能够组成的总球数的组合。
    它通过嵌套循环，遍历所有可能的红球和白球数量，计算对应的黑球数量，
    如果黑球数量在允许的范围内，则将该组合添加到结果列表中。

    参数:
    red (int): 红球的最大数量。
    white (int): 白球的最大数量。
    black (int): 黑球的最大数量。
    total (int): 组合中的总球数。

    返回:
    list: 包含所有可能的颜色组合的列表，每个组合是一个元组，包含红、白、黑球的数量。
    """

    # 初始化存储颜色组合的列表
    combinations = []

    # 遍历所有可能的红球数量
    for r in range(red + 1):
        # 遍历所有可能的白球数量
        for w in range(white + 1):
            # 计算黑球的数量
            b = total - r - w
            # 检查黑球的数量是否在允许的范围内
            if b >= 0 and b <= black:
                # 将符合条件的颜色组合添加到结果列表中
                combinations.append((r, w, b))

    # 返回所有可能的颜色组合
    return combinations

if __name__ == "__main__":
    red_balls = 3
    white_balls = 3
    black_balls = 6
    total_balls = 8

    all_combinations = count_color_combinations(red_balls, white_balls, black_balls, total_balls)
    print("共有", len(all_combinations), "种可能的颜色搭配：")
    print("红球 \t 白球 \t 黑球")
    print("........................")    
    for combination in all_combinations: 
        print(" %d \t %d \t %d" %( combination[0], combination[1], combination[2]))


